package com.dolphin.aiagentplatform.ai.constant;

public enum ModelTypeConst {

    DEEPSEEK_R1("deepseek-r1"),
    DEEPSEEK_V3("deepseek-v3"),
    QWEN_MAX("qwen-max");

    private final String modelType;

    ModelTypeConst(String modelType) {
        this.modelType = modelType;
    }

    public String getModelType() {
        return modelType;
    }

    public static ModelTypeConst getByModelType(String modelType) {
        for (ModelTypeConst modelTypeConst : values()) {
            if (modelTypeConst.getModelType().equalsIgnoreCase(modelType)) {
                return modelTypeConst;
            }
        }
        return null;
    }
} 